#include "testcase_utils.h"
#include <stdlib.h>

void ClearHcclEnv()
{
    unsetenv("HCCL_HIGH_PERF_ENABLE");
    unsetenv("HCCL_DETERMINISTIC");
    unsetenv("HCCL_INTRA_PCIE_ENABLE");
    unsetenv("HCCL_INTRA_ROCE_ENABLE");
    unsetenv("HCCL_ALGO");
    unsetenv("HCCL_BUFFSIZE");
    unsetenv("HCCL_OP_BASE_FFTS_MODE_ENABLE");
    unsetenv("HCCL_INTER_HCCS_DISABLE");
    unsetenv("HCCL_OP_EXPANSION_MODE");
    unsetenv("HCCL_CONCURRENT_ENABLE");
    return;
}

std::vector<u64> GenerateSendCountMatrix(u64 count, u32 rankSize)
{
    std::vector<u64> sendCountMatrix(rankSize * rankSize, count);
    return sendCountMatrix;
}
